Language Models Can Control Their Own Attention
语言模型可以控制自身注意力
🧭 LLM 长上下文稀疏注意力 ★ 深度解读
KAIST AI · Google DeepMind
提出声明式注意力协议,让模型在思维链中显式声明注意力范围,零训练实现长上下文稀疏注意力
📌 工作参考价值(4 点)
- 在长上下文多轮对话(电商客服、购物助手、推荐解释 Agent)中,把历史对话、商品信息、检索结果切分为可寻址片段,让模型通过 <focus> 标签声明需要关注的片段,可减少 50% 以上的 KV cache 读取,适合大 batch 在线服务降延迟。
- 与 RAG/工具调用结合:将检索到的商品/广告/知识片段作为模拟工具返回的“magic chunk”,模型在推理中自行选择关注哪些片段,避免每步全量扫描,可显著降低多文档推理成本。
- 工程上可直接复用论文的 vLLM 集成思路:通过 hook 改写 KV cache block table 实现块对齐掩码,不改内核、兼容 FlashAttention,适合快速在现有推荐/搜索 Agent 服务中落地。
- DA 的注意力计划以文本形式输出,可审计模型在推荐解释、搜索排序中具体关注了哪些输入,用于优化上下文构造和检索策略;未来还可结合 KV cache offloading 实现可逆上下文压缩,适合长会话 Agent。